package com.xiaozhen.security.exception;

import cn.dev33.satoken.exception.DisableLoginException;
import cn.dev33.satoken.exception.NotLoginException;
import cn.dev33.satoken.exception.NotPermissionException;
import cn.dev33.satoken.exception.NotRoleException;
import com.xiaozhen.common.core.domain.AjaxResult;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.bind.annotation.RestControllerAdvice;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

/**
 * HISTORY
 *
 * @author zhenweisheng
 * @date 2021/11/17 18:31
 * @description:
 */
@RestControllerAdvice
public class GlobalException {

    /**
     * 全局异常拦截（拦截项目中的所有异常）
     *
     * @param e
     * @param request
     * @param response
     * @return
     * @throws Exception
     */
    @ResponseBody
    @ExceptionHandler
    public AjaxResult handlerException(Exception e, HttpServletRequest request, HttpServletResponse response)
            throws Exception {

        // 打印堆栈，以供调试
        System.out.println("全局异常---------------");
        e.printStackTrace();

        // 不同异常返回不同状态码
        AjaxResult ajaxResult = null;
        if (e instanceof NotLoginException) {
            // 如果是未登录异常
            NotLoginException ee = (NotLoginException) e;
            ajaxResult = AjaxResult.error(ee.getMessage());
        } else if (e instanceof NotRoleException) {
            // 如果是角色异常
            NotRoleException ee = (NotRoleException) e;
            ajaxResult = AjaxResult.error("无此角色：" + ee.getRole());
        } else if (e instanceof NotPermissionException) {
            // 如果是权限异常
            NotPermissionException ee = (NotPermissionException) e;
            ajaxResult = AjaxResult.error("无此权限：" + ee.getCode());
        } else if (e instanceof DisableLoginException) {
            // 如果是被封禁异常
            DisableLoginException ee = (DisableLoginException) e;
            ajaxResult = AjaxResult.error("账号被封禁：" + ee.getDisableTime() + "秒后解封");
        }

        // 返回给前端
        return ajaxResult;
    }
}
